
#pragma once

#include  "hgeUtil.h"

class UnitSpawn
{
public:

	// Constructor
	UnitSpawn();

	// Sets the amount of a particular unit in a specific
	// wave. Automatically will resize the array
	void SetWaveUnit(int wave, int unit, int number);
	inline int GetWaveUnit(int wave, int unit)
	{ return wave < (int)waves.size() ? waves[wave][unit]: 0; }

	// Checks to see if a wave has units in it
	bool WaveHasUnits(int wave);

	// Waves units
	vector< vector<int> > waves;
};